package com.dhl.flyweight;

import java.util.HashMap;

/**
 * @title: WebSiteFactory
 * @Author danghailiang
 * @Date: 2022/8/16 12:09
 * @Version 1.0
 */
public  class WebSiteFactory {
    // 数据池
    private HashMap<String, ConcreteWebSite> pool = new HashMap<>();
    // 根据网站的类型返回一个网站，如果没有就创建一个网站，放入池中，并返回
    public WebSite getWebSiteCategory(String type){
        if(!pool.containsKey(type)){
            pool.put(type, new ConcreteWebSite(type));
        }
        return (WebSite) pool.get(type);
    }
    // 获取数量类型
    public int getWebSiteCount(){
        return pool.size();
    }
}
